home *** CD-ROM | disk | FTP | other *** search
/ Celestin Apprentice 5 / Apprentice-Release5.iso / Source Code / Libraries / Berkeley DB 1.8.5a / recno / rec_search.c < prev    next >
Text File  |  1995-06-22  |  4KB  |  132 lines
  1. /*-
  2.  * Copyright (c) 1990, 1993
  3.  *    The Regents of the University of California.  All rights reserved.
  4.  *
  5.  * Redistribution and use in source and binary forms, with or without
  6.  * modification, are permitted provided that the following conditions
  7.  * are met:
  8.  * 1. Redistributions of source code must retain the above copyright
  9.  *    notice, this list of conditions and the following disclaimer.
  10.  * 2. Redistributions in binary form must reproduce the above copyright
  11.  *    notice, this list of conditions and the following disclaimer in the
  12.  *    documentation and/or other materials provided with the distribution.
  13.  * 3. All advertising materials mentioning features or use of this software
  14.  *    must display the following acknowledgement:
  15.  *    This product includes software developed by the University of
  16.  *    California, Berkeley and its contributors.
  17.  * 4. Neither the name of the University nor the names of its contributors
  18.  *    may be used to endorse or promote products derived from this software
  19.  *    without specific prior written permission.
  20.  *
  21.  * THIS SOFTWARE IS PROVIDED BY THE REGENTS AND CONTRIBUTORS ``AS IS'' AND
  22.  * ANY EXPRESS OR IMPLIED WARRANTIES, INCLUDING, BUT NOT LIMITED TO, THE
  23.  * IMPLIED WARRANTIES OF MERCHANTABILITY AND FITNESS FOR A PARTICULAR PURPOSE
  24.  * ARE DISCLAIMED.  IN NO EVENT SHALL THE REGENTS OR CONTRIBUTORS BE LIABLE
  25.  * FOR ANY DIRECT, INDIRECT, INCIDENTAL, SPECIAL, EXEMPLARY, OR CONSEQUENTIAL
  26.  * DAMAGES (INCLUDING, BUT NOT LIMITED TO, PROCUREMENT OF SUBSTITUTE GOODS
  27.  * OR SERVICES; LOSS OF USE, DATA, OR PROFITS; OR BUSINESS INTERRUPTION)
  28.  * HOWEVER CAUSED AND ON ANY THEORY OF LIABILITY, WHETHER IN CONTRACT, STRICT
  29.  * LIABILITY, OR TORT (INCLUDING NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY
  30.  * OUT OF THE USE OF THIS SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF
  31.  * SUCH DAMAGE.
  32.  */
  33.  
  34. #if defined(LIBC_SCCS) && !defined(lint)
  35. static char sccsid[] = "@(#)rec_search.c    8.4 (Berkeley) 7/14/94";
  36. #endif /* LIBC_SCCS and not lint */
  37.  
  38. #include <sys/types.h>
  39.  
  40. #include <errno.h>
  41. #include <stdio.h>
  42.  
  43. #include <db.h>
  44. #include "recno.h"
  45.  
  46. /*
  47.  * __REC_SEARCH -- Search a btree for a key.
  48.  *
  49.  * Parameters:
  50.  *    t:    tree to search
  51.  *    recno:    key to find
  52.  *    op:     search operation
  53.  *
  54.  * Returns:
  55.  *    EPG for matching record, if any, or the EPG for the location of the
  56.  *    key, if it were inserted into the tree.
  57.  *
  58.  * Returns:
  59.  *    The EPG for matching record, if any, or the EPG for the location
  60.  *    of the key, if it were inserted into the tree, is entered into
  61.  *    the bt_cur field of the tree.  A pointer to the field is returned.
  62.  */
  63. #ifdef macintosh
  64. EPG *
  65. __rec_search(BTREE *t, recno_t recno, enum SRCHOP op)
  66. #else
  67. EPG *
  68. __rec_search(t, recno, op)
  69.     BTREE *t;
  70.     recno_t recno;
  71.     enum SRCHOP op;
  72. #endif
  73. {
  74.     register indx_t index;
  75.     register PAGE *h;
  76.     EPGNO *parent;
  77.     RINTERNAL *r;
  78.     pgno_t pg;
  79.     indx_t top;
  80.     recno_t total;
  81.     int sverrno;
  82.  
  83.     BT_CLR(t);
  84.     for (pg = P_ROOT, total = 0;;) {
  85.         if ((h = mpool_get(t->bt_mp, pg, 0)) == NULL)
  86.             goto err;
  87.         if (h->flags & P_RLEAF) {
  88.             t->bt_cur.page = h;
  89.             t->bt_cur.index = recno - total;
  90.             return (&t->bt_cur);
  91.         }
  92.         for (index = 0, top = NEXTINDEX(h);;) {
  93.             r = GETRINTERNAL(h, index);
  94.             if (++index == top || total + r->nrecs > recno)
  95.                 break;
  96.             total += r->nrecs;
  97.         }
  98.  
  99.         BT_PUSH(t, pg, index - 1);
  100.         
  101.         pg = r->pgno;
  102.         switch (op) {
  103.         case SDELETE:
  104.             --GETRINTERNAL(h, (index - 1))->nrecs;
  105.             mpool_put(t->bt_mp, h, MPOOL_DIRTY);
  106.             break;
  107.         case SINSERT:
  108.             ++GETRINTERNAL(h, (index - 1))->nrecs;
  109.             mpool_put(t->bt_mp, h, MPOOL_DIRTY);
  110.             break;
  111.         case SEARCH:
  112.             mpool_put(t->bt_mp, h, 0);
  113.             break;
  114.         }
  115.  
  116.     }
  117.     /* Try and recover the tree. */
  118. err:    sverrno = errno;
  119.     if (op != SEARCH)
  120.         while  ((parent = BT_POP(t)) != NULL) {
  121.             if ((h = mpool_get(t->bt_mp, parent->pgno, 0)) == NULL)
  122.                 break;
  123.             if (op == SINSERT)
  124.                 --GETRINTERNAL(h, parent->index)->nrecs;
  125.             else
  126.                 ++GETRINTERNAL(h, parent->index)->nrecs;
  127.                         mpool_put(t->bt_mp, h, MPOOL_DIRTY);
  128.                 }
  129.     errno = sverrno;
  130.     return (NULL);
  131. }
  132.